/*********************************************************************************
 *      Copyright:  (C) 2023 iot<iot@email.com>
 *                  All rights reserved.
 *
 *       Filename:  mai.c
 *    Description:  This file 
 *                 
 *        Version:  1.0.0(05/06/23)
 *         Author:  iot <iot@email.com>
 *      ChangeLog:  1, Release initial version on "05/06/23 15:59:38"
 *                 
 ********************************************************************************/
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <errno.h>
#include <ctype.h>
#include <time.h>
#include <signal.h>
#include <pthread.h>
#include <getopt.h>
#include <libgen.h>
#include <sys/types.h> 
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/epoll.h>
#include <sys/resource.h>

#include "logger.h"
#include "sql_serve.h"
#include "socket_server.h"

#define MAX_EVENTS 1024

static inline void print_usage(char *progname);
void sig_handle(int signum);

int		g_sigstop = 0;

int main(int argc,char **argv)
{
	int					listenfd, connfd;
	int					serv_port = 0;
	int					daemon_run = 0;
	char				*progname = NULL;
	int					opt;
	int					rv;
	int					i, j;
	int					found;
	char				buf[1024];
	int					epollfd;
	struct				epoll_event event;
	struct				epoll_event event_array[MAX_EVENTS];
	int					events;
	int					t = 0;
	char				*p = NULL;
	char				*buf_data[3];
	char				db_name[32] = "sql_temp_ser.db";

	struct option long_options[] =
	{ 
		{"daemon", no_argument, NULL, 'b'},
		{"port", required_argument, NULL, 'p'},
		{"help", no_argument, NULL, 'h'},
		{NULL, 0, NULL, 0}
	};

	progname = basename(argv[0]);

	/* Parser the command line parameters */
	while ((opt = getopt_long(argc, argv, "bp:h", long_options, NULL)) != -1)
	{ 
		switch (opt)
		{ 
			case 'b':
				daemon_run=1;
				break;
			case 'p':
				serv_port = atoi(optarg);
				break;
			case 'h': 
				print_usage(progname);
				return EXIT_SUCCESS;
			default:
				break;
		} 
	}
	
	if( !serv_port )
	{ 
		print_usage(progname);
		return -1;
	}
	
	/* Example Initialize the log system */
	if( logger_init("server.log",LOG_LEVEL_DEBUG) < 0 )
	{
		fprintf(stderr,"initial logger system failure\n");
		return -2;
	}
 
 	/* Mounting signal */
	signal(SIGINT,sig_handle);
	signal(SIGTERM,sig_handle);
 
 	/* creat database */
	if( creat_db(db_name) < 0 )
	{
		log_warning("creat database failure\n");
		return -3;
	}
	
	set_socket_rlimit(); /* set max open socket count */
	
	if( (listenfd = socket_server_init(NULL,serv_port)) < 0 )
	{
		log_error("error: %s server listen on port %d failure\n", argv[0],serv_port);
		return -5;
	}
	log_info("%s server start to listen on port %d\n", argv[0],serv_port);
   
	/* set program running on background */
	if( daemon_run )
	{
		if( daemon(0,0) )
		{
			log_error("running daemon failure:%s\n",strerror(errno));
		}
		log_info("running daemon successfully!\n");
	}
	
	if( (epollfd=epoll_create(MAX_EVENTS)) < 0 )
	{
		log_error("epoll_create() failure: %s\n", strerror(errno));
		return -6;
	}

	event.events = EPOLLIN;
	event.data.fd = listenfd;
	
	if( epoll_ctl(epollfd, EPOLL_CTL_ADD, listenfd, &event) < 0)
	{
		log_error("epoll add listen socket failure: %s\n", strerror(errno));
		return -7;
	}
	
	while(  !g_sigstop )
	{
		/* program will blocked here */ 
		events = epoll_wait(epollfd, event_array, MAX_EVENTS, -1);
		
		if(events < 0)
		{
			log_error("epoll failure: %s\n", strerror(errno));
			break;
		}
		else if(events == 0)
		{
			log_warning("epoll get timeout\n");
			continue;
		}
		
		/* rv>0 is the active events count */
		for(i=0; i<events; i++)
		{
			if ( (event_array[i].events&EPOLLERR) || (event_array[i].events&EPOLLHUP) )
			{
				log_error("epoll_wait get error on fd[%d]: %s\n",event_array[i].data.fd,strerror(errno));
				epoll_ctl(epollfd, EPOLL_CTL_DEL, event_array[i].data.fd, NULL);
				close(event_array[i].data.fd);
			}
			
			/* listen socket get event means new client start connect now */
			if( event_array[i].data.fd == listenfd )
			{
				if( (connfd=accept(listenfd, (struct sockaddr *)NULL, NULL)) < 0)
				{
					log_error("accept new client failure:%s\n",strerror(errno));
					continue;
				}
 
				event.data.fd = connfd;
				event.events = EPOLLIN;
				
				if( epoll_ctl(epollfd, EPOLL_CTL_ADD, connfd, &event) < 0 )
				{
					log_error("epoll add client socket failure: %s\n", strerror(errno));
					close(event_array[i].data.fd);
					continue;
				}
				log_info("epoll add new client socket[%d] ok.\n", connfd);
			}
			else /* already connected client socket get data incoming */
			{
				if( (rv=read(event_array[i].data.fd, buf, sizeof(buf))) <= 0)
				{
					log_error("socket[%d] read failure or get disconncet and will be removed.\n",event_array[i].data.fd);
					epoll_ctl(epollfd, EPOLL_CTL_DEL, event_array[i].data.fd, NULL);
					close(event_array[i].data.fd);
					continue;
				}
				else
				{
					log_info("socket[%d] read get %d bytes data\n",event_array[i].data.fd,rv);
					log_info("the data from client is:%s\n",buf);
				}
				
				p = strtok(buf,"|");
				
				while(p)
				{
					buf_data[t] = p;
					t += 1;
					p = strtok(NULL,"|");
				}

				log_info("A:%s  B:%s  c:%s\n",buf_data[0],buf_data[1],buf_data[2]);
				insert_db(buf_data);
			}
		} /* for(i=0; i<rv; i++) */
	} /* while(1) */

CleanUp:
	close(listenfd);
	
	return 0;
}

static inline void print_usage(char *progname)
{
	printf("usage: %s [OPTION]...\n",progname);

	printf(" %s is a socket server program,which used to verify client and echo back string from it\n",progname);
	printf("\nMandatory arguments to long options are mandatory for short options too:\n");

	printf(" -b[daemon ] set program running on background\n");
 	printf(" -p[port ] Socket server port address\n");
 	printf(" -h[help ] Display this help information\n");

	printf("\nExample: %s -b -p 8900\n", progname);
 	return ;
}

void sig_handle(int signum)
{
	log_warning("catch signal [%d]\n",signum);

	g_sigstop = 1;
}

